from gradio_client import Client, handle_file, file


def main(audio, image):
    """myhallo
    return: {'video': '/tmp/gradio/4b95d890bdad2f7255d558e319ca74a2bcb74356/output.mp4', 'subtitles': None}
    """
    client = Client("http://myhallo:3000/")
    result = client.predict(
        image=handle_file(image),
        audio=handle_file(audio),
        pose_weight=1,
        face_weight=1,
        lip_weight=1,
        face_expand_ratio=1.2,
        api_name="/predict",
    )
    print(result)
    return result
